import time
import cv2
import os
# 读取图片

def cut(name):
    image = cv2.imread('img_gray/'+name)
    cutname = []
    cutname.append(name[:1])
    cutname.append(name[1:2])
    cutname.append(name[2:3])
    cutname.append(name[3:4])
    # 获取图片的高度和宽度
    height, width = image.shape[:2]
    # 计算每一部分的宽度
    part_width = width // 4
    # 分割图片
    left = 0
    right = part_width
    for i in range(4):
        # 创建一个新的图片对象
        part = image[0:height, left:right]
        # 保存分割后的图片
        output_folder = 'img_cut/'+cutname[i]
        if not os.path.exists(output_folder):
            os.makedirs(output_folder)
        picname = time.time()
        cv2.imwrite(f'{output_folder}/{picname}.jpg', part)
        # 更新left和right的值
        left = right
        right += part_width
    # print('图片垂直方向分割完成')
    # print(name)

if __name__ == "__main__":
    # 取已经灰度的图片进行切割
    folder_path = 'img_gray'
    # 获取文件夹下所有文件的名字
    file_names = os.listdir(folder_path)
    # 输出文件名字
    for file_name in file_names:
        cut(file_name)